package com.burst.reptile.common.utils.Ffmpeg;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;

/**
 * 视频出来工具类
 * @author wujin
 *
 * @date: 2020年6月20日 下午6:18:52
 */
public class FfmpegUtil {
	
	/**
     * 截取视频封面
     * @param ffmpegPath 转码工具的路径
     * @param upFilePath 要截图的视频源文件 本地地址或者是http://
     * @param mediaPicPath 截图保存路径
     * @param second 截图第几秒的
     * @return
     * @throws Exception
     */
    public static void imgCover(String ffmpegPath, String upFilePath, String mediaPicPath, int second) {
    	String videoInfo = videoInfo(upFilePath);
    	JSONObject parseObject = JSONObject.parseObject(videoInfo);
    	JSONArray jsonArray = parseObject.getJSONArray("streams");
    	JSONObject jsonObject = jsonArray.getJSONObject(0);
    	Integer width = jsonObject.getInteger("width");
    	Integer height = jsonObject.getInteger("height");
//    	Integer width = 1280;
//    	Integer height = 720;
        // 创建一个List集合来保存从视频中截取图片的命令
        List<String> cutpic = new ArrayList<String>();
        cutpic.add(ffmpegPath);
        cutpic.add("-i");
        cutpic.add(upFilePath); // 同上（指定的文件即可以是转换为flv格式之前的文件，也可以是转换的flv文件）
        cutpic.add("-y");
        cutpic.add("-f");
        cutpic.add("image2");
        cutpic.add("-ss"); // 添加参数＂-ss＂，该参数指定截取的起始时间
        cutpic.add(second + ""); // 添加起始时间为第2秒
        cutpic.add("-t"); // 添加参数＂-t＂，该参数指定持续时间
        cutpic.add("0.001"); // 添加持续时间为1毫秒
        cutpic.add("-s"); // 添加参数＂-s＂，该参数指定截取的图片大小
        cutpic.add(width + "*" + height); // 添加截取的图片大小为350*240
        cutpic.add(mediaPicPath); // 添加截取的图片的保存路径
 
        ProcessBuilder builder = new ProcessBuilder();
        InputStream is = null;
        InputStreamReader inst = null;
        BufferedReader br = null;
        try {
            builder.command(cutpic);
            builder.redirectErrorStream(true); // 如果此属性为 true，则任何由通过此对象的 start() 方法启动的后续子进程生成的错误输出都将与标准输出合并，
            Process process = builder.start(); //因此两者均可使用 Process.getInputStream() 方法读取。这使得关联错误消息和相应的输出变得更容易
            is = process.getInputStream();
            inst = new InputStreamReader(is, "GBK");
            br = new BufferedReader(inst);//输入流缓冲区
            String res = null;
        	StringBuilder sb = new StringBuilder();
        	while ((res = br.readLine()) != null) {//循环读取缓冲区中的数据
        	    sb.append(res+"\n");
        	}
        	process.destroy();
        } catch (Exception e) {
            e.printStackTrace();
        }finally {
        	//关闭流。自己记得关
        	if(is != null) {
        		try {
					is.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
        	}
        	if(inst != null) {
        		try {
					inst.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
        	}
        	if(br != null) {
        		try {
					br.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
        	}
        }
    }
    
    /**
     * 获取视频元数据信息
     * @param url 视频地址 本地或者是http://
     * @return
     */
    public static String videoInfo (String url) {
    	List<String> cutpic = new ArrayList<String>();
        cutpic.add("ffprobe");
        cutpic.add("-select_streams");
        cutpic.add("v"); 
        cutpic.add("-show_entries");
        cutpic.add("format=duration,size,bit_rate,filename");
        cutpic.add("-show_streams");
        cutpic.add("-v"); 
        cutpic.add("quiet"); 
        cutpic.add("-of"); 
        cutpic.add("csv=\"p=0\""); 
        cutpic.add("-of"); 
        cutpic.add("json"); 
        cutpic.add("-i"); 
        cutpic.add(url);
        ProcessBuilder builder = new ProcessBuilder();
        InputStream is = null;
        InputStreamReader inst = null;
        BufferedReader br = null;
        try {
            builder.command(cutpic);
            builder.redirectErrorStream(true); // 如果此属性为 true，则任何由通过此对象的 start() 方法启动的后续子进程生成的错误输出都将与标准输出合并，
            Process process = builder.start(); //因此两者均可使用 Process.getInputStream() 方法读取。这使得关联错误消息和相应的输出变得更容易
            is = process.getInputStream();
            inst = new InputStreamReader(is, "GBK");
            br = new BufferedReader(inst);//输入流缓冲区
            String res = null;
        	StringBuilder sb = new StringBuilder();
        	while ((res = br.readLine()) != null) {//循环读取缓冲区中的数据
        	    sb.append(res+"\n");
        	}
        	process.destroy();
            return sb.toString();
        } catch (Exception e) {
            e.printStackTrace();
        }finally {
        	if(is != null) {
        		try {
					is.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
        	}
        	if(inst != null) {
        		try {
					inst.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
        	}
        	if(br != null) {
        		try {
					br.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
        	}
        }
    	return "";
    }
    
    
    /**
     * 视频转码
     * @param ffmpegPath    转码工具的存放路径
     * @param upFilePath    用于指定要转换格式的文件,要截图的视频源文件
     * @param codcFilePath    格式转换后的的文件保存路径
     * @return
     * @throws Exception
     */
    public static void executeCodecs(String ffmpegPath, String upFilePath, String codcFilePath) {
        // 创建一个List集合来保存转换视频文件为flv格式的命令
        List<String> convert = new ArrayList<String>();
        convert.add(ffmpegPath); // 添加转换工具路径
        convert.add("-i"); // 添加参数＂-i＂，该参数指定要转换的文件
        convert.add(upFilePath); // 添加要转换格式的视频文件的路径
//        convert.add("-qscale");     //指定转换的质量
//        convert.add("6");
//        convert.add("-ab");        //设置音频码率
//        convert.add("64");
//        convert.add("-ac");        //设置声道数
//        convert.add("2");
//        convert.add("-ar");        //设置声音的采样频率
//        convert.add("22050");
//        convert.add("-r");        //设置帧频
//        convert.add("24");
//        convert.add("-y"); // 添加参数＂-y＂，该参数指定将覆盖已存在的文件
//        convert.add(codcFilePath);
        convert.add("-profile:v");
        convert.add("baseline");
        convert.add("-level");
        convert.add("3.0");
        convert.add("-s");
        convert.add("640x360");
        convert.add("-start_number");
        convert.add("0");
        convert.add("-hls_time");
        convert.add("10");
        convert.add("-hls_list_size");
        convert.add("0");
        convert.add("-f");
        convert.add("hls");
        convert.add(codcFilePath);
        
        ProcessBuilder builder = new ProcessBuilder();
        InputStream is = null;
        InputStreamReader inst = null;
        BufferedReader br = null;
        try {
            builder.command(convert);
            builder.redirectErrorStream(true);
            builder.start();
            // 如果此属性为 true，则任何由通过此对象的 start() 方法启动的后续子进程生成的错误输出都将与标准输出合并，
            //因此两者均可使用 Process.getInputStream() 方法读取。这使得关联错误消息和相应的输出变得更容易
            Process process = builder.start();
            is = process.getInputStream();
            inst = new InputStreamReader(is, "GBK");
            br = new BufferedReader(inst);//输入流缓冲区
            String res = null;
        	StringBuilder sb = new StringBuilder();
        	while ((res = br.readLine()) != null) {//循环读取缓冲区中的数据
        	    sb.append(res+"\n");
        	}
        	process.destroy();
        } catch (Exception e) {
            e.printStackTrace();
        }finally {
        	if(is != null) {
        		try {
					is.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
        	}
        	if(inst != null) {
        		try {
					inst.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
        	}
        	if(br != null) {
        		try {
					br.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
        	}
        }
    }
    
    public static void main(String orgs[]) throws Exception {
    	imgCover("D:/tool/ffmpeg/ffmpeg-20200113-7225479-win64-static/bin/ffmpeg.exe", 
    			"http://bpic.588ku.com/video_listen/588ku_video/20/06/11/17/45/28/video5ee1fd387db9a.mp4", 
    			"D:/piecemeal/333/12346.jpg",
    			1);
//    	String dd = videoInfo("https://sxp-1259003793.cos.ap-chongqing.myqcloud.com/video/50fb121ed7d84d0cab26de7068c31f29.mp4");
//    	System.out.println(dd);
    }
    
}
